# 在本实例中我们比较阶跃函数和SIGMOD函数
# 在同一张图中画出这两个函数的图像
import matplotlib.pyplot as plt
import numpy as np


def step_function(x):  # 阶跃函数的返回值只能是0或者1
    y = x > 0
    y = y.astype(np.int)
    return y


def sigmod_function(x):
    y = 1 / (1 + np.exp(-x))
    return y


# 函数入口
X = np.arange(-5, 5, 0.1)  # 将需要的参数全部设置好
Y1 = step_function(X)
Y2 = sigmod_function(X)

plt.plot(X, Y1, label='step')  # 开始绘画
plt.plot(X, Y2, linestyle='--', label='sigmod')
plt.xlabel('X')
plt.ylabel('Y')
plt.legend()
plt.title('two functions compare')
plt.ylim(-0.1, 1.1)
plt.show()
